package com.tuniu.agents.common;

public enum TraceEnum {

    ORIGINAL_INPUT(1,"Original user input"),
    REWRITE_INPUT(2,"Rewritten input"),
    AGENT_INPUT(3,"Input when calling Agent"),
    LLM_INPUT(4,"Input when calling LLM"),
    TOOL_INPUT(5,"Input when calling a tool"),
    TOOL_OUTPUT(6,"Output when calling a tool"),
    LLM_OUTPUT(7,"Output when calling LLM"),
    AGENT_OUTPUT(8,"Output when calling Agent"),
    FINAL_OUTPUT(9,"Final output");

    private Integer code; // Code representing the trace type
    private String description; // Description of the trace type

    TraceEnum(Integer code,String description){
        this.code = code; // Initialize code
        this.description = description; // Initialize description
    }

    public Integer getCode() {
        return code; // Get the code
    }

    public String getDescription() {
        return description; // Get the description
    }
}
